package com.zareal.ymcc.service.impl;

import com.baomidou.mybatisplus.mapper.EntityWrapper;
import com.baomidou.mybatisplus.mapper.Wrapper;
import com.baomidou.mybatisplus.service.impl.ServiceImpl;
import com.zareal.ymcc.constans.RedisConstants;
import com.zareal.ymcc.domain.CourseType;
import com.zareal.ymcc.dto.CourseTypeCrumbsDto;
import com.zareal.ymcc.enums.GlobalErrorCode;
import com.zareal.ymcc.mapper.CourseTypeMapper;
import com.zareal.ymcc.service.ICourseTypeService;
import com.zareal.ymcc.util.AssertUtil;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.stereotype.Service;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * <p>
 * 课程目录 服务实现类
 * </p>
 *
 * @author Wilson
 * @since 2022-09-30
 */
@Service
public class CourseTypeServiceImpl extends ServiceImpl<CourseTypeMapper, CourseType> implements ICourseTypeService {

    @Override
    @Cacheable(cacheNames = RedisConstants.SPRING_CACHE_COURSE_TYPE_TREE, key = "'TREE'")
    public List<CourseType> treeData() {
        return tree();
    }

    /**
     * 面包屑查询
     *
     * @param courseTypeId
     * @return
     */
    @Override
    public List<CourseTypeCrumbsDto> crumbs(Long courseTypeId) {
        CourseType courseType = selectById(courseTypeId);
        AssertUtil.isNotNull(courseType, GlobalErrorCode.COURSE_TYPE_IS_NULL);
        List<CourseTypeCrumbsDto> crumbsDtoList = new ArrayList<>();
        String[] split = courseType.getPath().split("\\.");
// 设置一个面包屑返回
        for (String s : split) {
            CourseTypeCrumbsDto courseTypeCrumbsDto = new CourseTypeCrumbsDto();
            CourseType courseType1 = selectById(Long.valueOf(s));
            courseTypeCrumbsDto.setOwnerProductType(courseType1);
            Long pid = courseType1.getPid();
            Wrapper<CourseType> wrapper = new EntityWrapper<>();
            wrapper.eq("pid",pid);
            courseTypeCrumbsDto.setOtherProductTypes(selectList(wrapper));
            crumbsDtoList.add(courseTypeCrumbsDto);
        }

    return crumbsDtoList;
}

    @Override
    // 会在数据库更新之后，更新缓存数据
    @CacheEvict(cacheNames = RedisConstants.SPRING_CACHE_COURSE_TYPE_TREE, key = "'TREE'")
    public boolean insert(CourseType entity) {
        return super.insert(entity);
    }


    @Override
    // 会在数据库更新之后，更新缓存数据
    @CacheEvict(cacheNames = RedisConstants.SPRING_CACHE_COURSE_TYPE_TREE, key = "'TREE'")
    public boolean updateById(CourseType entity) {
        return super.updateById(entity);
    }


    @Override
    // 会在数据库更新之后，更新缓存数据
    @CacheEvict(cacheNames = RedisConstants.SPRING_CACHE_COURSE_TYPE_TREE, key = "'TREE'")
    public boolean deleteById(Serializable id) {
        return super.deleteById(id);
    }

    private ArrayList<CourseType> tree() {
        // 查询所有数据
        List<CourseType> courseTypes = selectList(null);
        // 转换为map集合
        Map<Long, CourseType> collect = courseTypes.stream().collect(Collectors.toMap(CourseType::getId, item -> item));
        // 遍历获取数据
        ArrayList<CourseType> treeData = new ArrayList<>();

        for (CourseType courseType : courseTypes) {
            // 查询pid是不是空或者不存在
            if (courseType.getPid() == null || courseType.getPid().intValue() == 0) {
                treeData.add(courseType);
            } else {
                // 存在就是有爹
                collect.get(courseType.getPid()).getChildren().add(courseType);
            }
        }

        return treeData;
    }
}
